from model.uml_base import UmlBase
from model.uml_xmi_info import UmlXmiInfo


class UmlComment(UmlBase):
    def __init__(self) -> None:
        super().__init__(UmlXmiInfo.type_comment)
        self.body: str = ""

        """
        id ref
        """
        self.annotatedElement: str = ""

    def to_json(self) -> dict[str, object]:
        json: dict = super().to_json()
        json["body"] = self.body
        json["annotatedElement"] = self.annotatedElement

        return json
